UnitVectorLayer

UnitVectorLayer[n] represents a net layer that transforms integers between 1 and n into n-dimensional unit vectors. 

UnitVectorLayer[] leaves the n to be inferred from context.

Please visit the official Wolfram Language Reference for more details and examples on core symbols.